#include "stdafx.h"
#if defined(__WIN32__)
#include <cc/core/windows/windows.h>
#endif

/* CPU feature detection for CCLIB */

#include "cpu_info.h"

#ifdef HAVE_SYSCONF
#include <unistd.h>
#endif
#ifdef HAVE_SYSCTLBYNAME
#include <sys/types.h>
#include <sys/sysctl.h>
#endif
#if defined(__MACOSX__) && (defined(__ppc__) || defined(__ppc64__))
#include <sys/sysctl.h> /* For AltiVec check */
#elif defined(__OpenBSD__) && defined(__powerpc__)
#include <sys/param.h>
#include <sys/sysctl.h> /* For AltiVec check */
#include <machine/cpu.h>
#elif CC_ALTIVEC_BLITTERS && HAVE_SETJMP
#include <signal.h>
#include <setjmp.h>
#endif

#if CC_ALTIVEC_BLITTERS && HAVE_SETJMP && !__MACOSX__ && !__OpenBSD__
/* This is the brute force way of detecting instruction sets...
   the idea is borrowed from the libmpeg2 library - thanks!
 */
static jmp_buf jmpbuf;
static void  illegal_instruction(int sig)
{
    longjmp(jmpbuf, 1);
}
#endif /* HAVE_SETJMP */

static inline int  CPU_haveCPUID(void)
{
    int has_CPUID = 0;
/* *INDENT-OFF* */
#if defined(__GNUC__) && defined(i386)
    __asm__ (
"        pushfl                      # Get original EFLAGS             \n"
"        popl    %%eax                                                 \n"
"        movl    %%eax,%%ecx                                           \n"
"        xorl    $0x200000,%%eax     # Flip ID bit in EFLAGS           \n"
"        pushl   %%eax               # Save new EFLAGS value on stack  \n"
"        popfl                       # Replace current EFLAGS value    \n"
"        pushfl                      # Get new EFLAGS                  \n"
"        popl    %%eax               # Store new EFLAGS in EAX         \n"
"        xorl    %%ecx,%%eax         # Can not toggle ID bit,          \n"
"        jz      1f                  # Processor=80486                 \n"
"        movl    $1,%0               # We have CPUID support           \n"
"1:                                                                    \n"
    : "=m" (has_CPUID)
    :
    : "%eax", "%ecx"
    );
#elif defined(__GNUC__) && defined(__x86_64__)
/* Technically, if this is being compiled under __x86_64__ then it has 
   CPUid by definition.  But it's nice to be able to prove it.  :)      */
    __asm__ (
"        pushfq                      # Get original EFLAGS             \n"
"        popq    %%rax                                                 \n"
"        movq    %%rax,%%rcx                                           \n"
"        xorl    $0x200000,%%eax     # Flip ID bit in EFLAGS           \n"
"        pushq   %%rax               # Save new EFLAGS value on stack  \n"
"        popfq                       # Replace current EFLAGS value    \n"
"        pushfq                      # Get new EFLAGS                  \n"
"        popq    %%rax               # Store new EFLAGS in EAX         \n"
"        xorl    %%ecx,%%eax         # Can not toggle ID bit,          \n"
"        jz      1f                  # Processor=80486                 \n"
"        movl    $1,%0               # We have CPUID support           \n"
"1:                                                                    \n"
    : "=m" (has_CPUID)
    :
    : "%rax", "%rcx"
    );
#elif (defined(_MSC_VER) && defined(_M_IX86)) || defined(__WATCOMC__)
    __asm {
        pushfd                      ; Get original EFLAGS
        pop     eax
        mov     ecx, eax
        xor     eax, 200000h        ; Flip ID bit in EFLAGS
        push    eax                 ; Save new EFLAGS value on stack
        popfd                       ; Replace current EFLAGS value
        pushfd                      ; Get new EFLAGS
        pop     eax                 ; Store new EFLAGS in EAX
        xor     eax, ecx            ; Can not toggle ID bit,
        jz      done                ; Processor=80486
        mov     has_CPUID,1         ; We have CPUID support
done:
    }
#elif defined(_MSC_VER) && defined(_M_X64)
    has_CPUID = 1;
#elif defined(__sun) && defined(__i386)
    __asm (
"       pushfl                 \n"
"       popl    %eax           \n"
"       movl    %eax,%ecx      \n"
"       xorl    $0x200000,%eax \n"
"       pushl   %eax           \n"
"       popfl                  \n"
"       pushfl                 \n"
"       popl    %eax           \n"
"       xorl    %ecx,%eax      \n"
"       jz      1f             \n"
"       movl    $1,-8(%ebp)    \n"
"1:                            \n"
    );
#elif defined(__sun) && defined(__amd64)
    __asm (
"       pushfq                 \n"
"       popq    %rax           \n"
"       movq    %rax,%rcx      \n"
"       xorl    $0x200000,%eax \n"
"       pushq   %rax           \n"
"       popfq                  \n"
"       pushfq                 \n"
"       popq    %rax           \n"
"       xorl    %ecx,%eax      \n"
"       jz      1f             \n"
"       movl    $1,-8(%rbp)    \n"
"1:                            \n"
    );
#endif
/* *INDENT-ON* */
    return has_CPUID;
}

#if defined(__GNUC__) && defined(i386)
#define cpuid(func, a, b, c, d) \
    __asm__ __volatile__ ( \
"        pushl %%ebx        \n" \
"        cpuid              \n" \
"        movl %%ebx, %%esi  \n" \
"        popl %%ebx         \n" : \
"=a" (a), "=S" (b), "=c" (c), "=d" (d) : "a" (func))
#elif defined(__GNUC__) && defined(__x86_64__)
#define cpuid(func, a, b, c, d) \
    __asm__ __volatile__ ( \
"        pushq %%rbx        \n" \
"        cpuid              \n" \
"        movq %%rbx, %%rsi  \n" \
"        popq %%rbx         \n" : \
"=a" (a), "=S" (b), "=c" (c), "=d" (d) : "a" (func))
#elif (defined(_MSC_VER) && defined(_M_IX86)) || defined(__WATCOMC__)
#define cpuid(func, a, b, c, d) \
        __asm { \
        __asm mov eax, func \
        __asm cpuid \
        __asm mov a, eax \
        __asm mov b, ebx \
        __asm mov c, ecx \
        __asm mov d, edx \
}
#elif defined(_MSC_VER) && defined(_M_X64)
#define cpuid(func, a, b, c, d) \
{ \
    int CPUInfo[4]; \
    __cpuid(CPUInfo, func); \
    a = CPUInfo[0]; \
    b = CPUInfo[1]; \
    c = CPUInfo[2]; \
    d = CPUInfo[3]; \
}
#else
#define cpuid(func, a, b, c, d) \
    a = b = c = d = 0
#endif

static inline int  CPU_GetCPUIDFeatures(void)
{
    int features = 0;
    int a, b, c, d;

    cpuid(0, a, b, c, d);
    if (a >= 1) {
        cpuid(1, a, b, c, d);
        features = d;
    }
    return features;
}

static inline int  CPU_haveRDTSC(void)
{
    if (CPU_haveCPUID()) {
        return (CPU_GetCPUIDFeatures() & 0x00000010);
    }
    return 0;
}

static inline int  CPU_haveAltiVec(void)
{
    volatile int altivec = 0;
#if (defined(__MACOSX__) && (defined(__ppc__) || defined(__ppc64__))) || (defined(__OpenBSD__) && defined(__powerpc__))
#ifdef __OpenBSD__
    int selectors[2] = { CTL_MACHDEP, CPU_ALTIVEC };
#else
    int selectors[2] = { CTL_HW, HW_VECTORUNIT };
#endif
    int hasVectorUnit = 0;
    size_t length = sizeof(hasVectorUnit);
    int error = sysctl(selectors, 2, &hasVectorUnit, &length, NULL, 0);
    if (0 == error)
        altivec = (hasVectorUnit != 0);
#elif CC_ALTIVEC_BLITTERS && HAVE_SETJMP
    void (*handler) (int sig);
    handler = signal(SIGILL, illegal_instruction);
    if (setjmp(jmpbuf) == 0) {
        asm volatile ("mtspr 256, %0\n\t" "vand %%v0, %%v0, %%v0"::"r" (-1));
        altivec = 1;
    }
    signal(SIGILL, handler);
#endif
    return altivec;
}

static inline int  CPU_haveMMX(void)
{
    if (CPU_haveCPUID()) {
        return (CPU_GetCPUIDFeatures() & 0x00800000);
    }
    return 0;
}

static inline int  CPU_have3DNow(void)
{
    if (CPU_haveCPUID()) {
        int a, b, c, d;

        cpuid(0x80000000, a, b, c, d);
        if (a >= 0x80000001) {
            cpuid(0x80000001, a, b, c, d);
            return (d & 0x80000000);
        }
    }
    return 0;
}

static inline int  CPU_haveSSE(void)
{
    if (CPU_haveCPUID()) {
        return (CPU_GetCPUIDFeatures() & 0x02000000);
    }
    return 0;
}

static inline int  CPU_haveSSE2(void)
{
    if (CPU_haveCPUID()) {
        return (CPU_GetCPUIDFeatures() & 0x04000000);
    }
    return 0;
}

static inline int  CPU_haveSSE3(void)
{
    if (CPU_haveCPUID()) {
        int a, b, c, d;

        cpuid(0, a, b, c, d);
        if (a >= 1) {
            cpuid(1, a, b, c, d);
            return (c & 0x00000001);
        }
    }
    return 0;
}

static inline int  CPU_haveSSE41(void)
{
    if (CPU_haveCPUID()) {
        int a, b, c, d;

        cpuid(1, a, b, c, d);
        if (a >= 1) {
            cpuid(1, a, b, c, d);
            return (c & 0x00080000);
        }
    }
    return 0;
}

static inline int  CPU_haveSSE42(void)
{
    if (CPU_haveCPUID()) {
        int a, b, c, d;

        cpuid(1, a, b, c, d);
        if (a >= 1) {
            cpuid(1, a, b, c, d);
            return (c & 0x00100000);
        }
    }
    return 0;
}

static inline int  CPU_haveAVX(void)
{
    if (CPU_haveCPUID()) {
        int a, b, c, d;

        cpuid(1, a, b, c, d);
        if (a >= 1) {
            cpuid(1, a, b, c, d);
            return (c & 0x10000000);
        }
    }
    return 0;
}

static int _cc_cpu_count = 0;

int c_cpu_count(void)
{
    if (!_cc_cpu_count) {
#if defined(HAVE_SYSCONF) && defined(_SC_NPROCESSORS_ONLN)
        if (_cc_cpu_count <= 0) {
            _cc_cpu_count = (int)sysconf(_SC_NPROCESSORS_ONLN);
        }
#endif
#ifdef HAVE_SYSCTLBYNAME
        if (_cc_cpu_count <= 0) {
            size_t size = sizeof(_cc_cpu_count);
            sysctlbyname("hw.ncpu", &_cc_cpu_count, &size, NULL, 0);
        }
#endif
#ifdef __WIN32__
        if (SDL_CPUCount <= 0) {
            SYSTEM_INFO info;
            GetSystemInfo(&info);
            _cc_cpu_count = info.dwNumberOfProcessors;
        }
#endif
        /* There has to be at least 1, right? :) */
        if (_cc_cpu_count <= 0) {
            _cc_cpu_count = 1;
        }
    }
    return _cc_cpu_count;
}

const TCHAR* c_cpu_sn()
{
    static TCHAR _cc_cpu_sn[32] = {0};
    int a, b, c, d;

    if (_cc_cpu_sn[0])
        return _cc_cpu_sn;

    if (CPU_haveCPUID()) {
        cpuid(0x00000000, a, b, c, d);
        _sntprintf(_cc_cpu_sn, _countof(_cc_cpu_sn), _T("%04X-%04X-%04X-%04X-%04X-%04X"),
            a>> 16, a & 0xffff,
            d>> 16, d & 0xffff,
            c>> 16, c & 0xffff);
    }

    return _cc_cpu_sn;
}

/* Oh, such a sweet sweet trick, just not very useful. :) */
const TCHAR * c_cpu_type(void)
{
    static TCHAR _cc_cpu_type[13] = {0};

    if (!_cc_cpu_type[0]) {
        int i = 0;
        int a, b, c, d;

        if (CPU_haveCPUID()) {
            cpuid(0x00000000, a, b, c, d);
            _cc_cpu_type[i++] = (TCHAR)(b & 0xff); b >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(b & 0xff); b >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(b & 0xff); b >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(b & 0xff); b >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(d & 0xff); d >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(d & 0xff); d >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(d & 0xff); d >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(d & 0xff); d >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(c & 0xff); c >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(c & 0xff); c >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(c & 0xff); c >>= 8;
            _cc_cpu_type[i++] = (TCHAR)(c & 0xff); c >>= 8;
        }
        if (!_cc_cpu_type[0]) {
            _tcscpy(_cc_cpu_type, _T("Unknown"));
        }
    }
    return _cc_cpu_type;
}


const TCHAR * c_cpu_name(void)
{
    static TCHAR _c_cpu_name[48] = {0};

    if (!_c_cpu_name[0]) {
        int i = 0;
        int a, b, c, d;

        if (CPU_haveCPUID()) {
            cpuid(0x80000000, a, b, c, d);
            if (a >= 0x80000004) {
                cpuid(0x80000002, a, b, c, d);
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                cpuid(0x80000003, a, b, c, d);
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                cpuid(0x80000004, a, b, c, d);
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(a & 0xff); a >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(b & 0xff); b >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(c & 0xff); c >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
                _c_cpu_name[i++] = (TCHAR)(d & 0xff); d >>= 8;
            }
        }
        if (!_c_cpu_name[0]) {
            _tcscpy(_c_cpu_name, _T("Unknown"));
        }
    }
    return _c_cpu_name;
}

int  c_cpu_cache_line_size(void)
{
    const TCHAR *cpuType = c_cpu_type();

    if (_tcsicmp(cpuType, _T("GenuineIntel")) == 0) {
        int a, b, c, d;

        cpuid(0x00000001, a, b, c, d);
        return (((b >> 8) & 0xff) * 8);
    } else if (_tcsicmp(cpuType, _T("AuthenticAMD")) == 0) {
        int a, b, c, d;

        cpuid(0x80000005, a, b, c, d);
        return (c & 0xff);
    } else {
        /* Just make a guess here... */
        return CC_CACHELINE_SIZE;
    }
}

static UINT32 _c_cpu_features = 0xFFFFFFFF;

UINT32  c_cpu_features(void)
{
    if (_c_cpu_features == 0xFFFFFFFF) {
        _c_cpu_features = 0;
        if (CPU_haveRDTSC()) {
            _c_cpu_features |= CPU_HAS_RDTSC;
        }
        if (CPU_haveAltiVec()) {
            _c_cpu_features |= CPU_HAS_ALTIVEC;
        }
        if (CPU_haveMMX()) {
            _c_cpu_features |= CPU_HAS_MMX;
        }
        if (CPU_have3DNow()) {
            _c_cpu_features |= CPU_HAS_3DNOW;
        }
        if (CPU_haveSSE()) {
            _c_cpu_features |= CPU_HAS_SSE;
        }
        if (CPU_haveSSE2()) {
            _c_cpu_features |= CPU_HAS_SSE2;
        }
        if (CPU_haveSSE3()) {
            _c_cpu_features |= CPU_HAS_SSE3;
        }
        if (CPU_haveSSE41()) {
            _c_cpu_features |= CPU_HAS_SSE41;
        }
        if (CPU_haveSSE42()) {
            _c_cpu_features |= CPU_HAS_SSE42;
        }
        if (CPU_haveAVX()) {
            _c_cpu_features |= CPU_HAS_AVX;
        }
    }
    return _c_cpu_features;
}

static int _cc_system_RAM = 0;

int c_system_RAM(void)
{
    if (!_cc_system_RAM) {
#if defined(HAVE_SYSCONF) && defined(_SC_PHYS_PAGES) && defined(_SC_PAGESIZE)
        if (_cc_system_RAM <= 0) {
            _cc_system_RAM = (int)((int64_t)sysconf(_SC_PHYS_PAGES) * sysconf(_SC_PAGESIZE) / (1024*1024));
        }
#endif
#ifdef HAVE_SYSCTLBYNAME
        if (_cc_system_RAM <= 0) {
#if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
#ifdef HW_REALMEM
            int mib[2] = {CTL_HW, HW_REALMEM};
#else
            /* might only report up to 2 GiB */
            int mib[2] = {CTL_HW, HW_PHYSMEM};
#endif /* HW_REALMEM */
#else
            int mib[2] = {CTL_HW, HW_MEMSIZE};
#endif /* __FreeBSD__ || __FreeBSD_kernel__ */
            uint64_t memsize = 0;
            size_t len = sizeof(memsize);
            
            if (sysctl(mib, 2, &memsize, &len, NULL, 0) == 0) {
                _cc_system_RAM = (int)(memsize / (1024*1024));
            }
        }
#endif
#ifdef __WIN32__
        if (_cc_system_RAM <= 0) {
            MEMORYSTATUSEX stat;
            stat.dwLength = sizeof(stat);
            if (GlobalMemoryStatusEx(&stat)) {
                _cc_system_RAM = (int)(stat.ullTotalPhys / (1024 * 1024));
            }
        }
#endif
    }
    return _cc_system_RAM;
}
